@MaksGal This unfortunately doesn't work in iOS 16 and later. If anyone finds a solution, I'd love to hear about it.

Does this approach still work for you? I try it on a tvOS 17 simulator, and it always resets to the first item, even if I say that the default focus index should be 2.

I've had people reaching out to me that their keyboard extensions started disappearing in iOS 17. As I investigated one, I found that using "se." (Swedish top domain) as bundle identifier prefix causes the keyboard to disappear. I wrote more about this issue here, and have reported it to the Feedback Assistant: https://keyboardkit.com/blog/2024/02/12/keyboards-disappear-in-ios-17

I have the same problem, and I've also reported it with the Feedback Assistant. Hopefully they get around to fixing this before 16-4, since it's quite crippling.